# 练习2：使用python实现tree命令
import os
def tree(path, prefix=''):
    print(prefix + os.path.basename(path) + '/')
    if os.path.isdir(path):
        for item in os.listdir(path):
            item_path = os.path.join(path, item)
            if os.path.isdir(item_path):
                tree(item_path, prefix=prefix + '  ')
            else:
                print(prefix + '        ' + item)

print("实现tree命令...")
addr = input("请输入你要tree的地址>>")
tree(addr)
